from calculator import Count
import unittest, ddt


@ddt.ddt
class TestCount(unittest.TestCase):
    # @classmethod. 如果想要在所有case执行之前准备一次环境，并在所有case执行结束之后再清理环境，
    # 可以用 setUpClass() 与tearDownClass()  https://www.cnblogs.com/fennudexiaoniao/p/7771931.html
    # 测试的初始化
    def setUp(self):
        print("test start")

    # 测试用例add的编写
    @ddt.data([1, 2, 3], [20, 12, 2134576980], [2, 3, 5])
    @ddt.unpack
    def test_add(self, a, b, c):
        self.Count = Count(a, b)
        self.assertEqual(self.Count.add(), c)

    # 测试的回收-测试的环境还原
    def tearDown(self):
        print("test end")


if __name__ == "__main__":
    unittest.main()
